Python:function annotation